package keter.web.util;

import org.dom4j.Document;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.SAXValidator;
import org.dom4j.io.XMLWriter;
import org.dom4j.util.XMLErrorHandler;
import org.joda.time.DateTime;

import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import java.io.File;

/**
* Java XML校验测试
*
* @author leizhimin，2008-9-4 14:42:35
*/
public class XmlXsdValidator {
    public static void main(String[] args) {
//        String xmlFileName = "E://git/keter/keter-project/web/src/main/webapp/WEB-INF/web.xml";
        String xmlFileName = "E:/WorkFolder/DevTools/ide/Workplace/keter/project/keter-project/web/src/main/webapp/WEB-INF/web.xml";
//        String xsdFileName = "E://git/keter/keter-project/web/src/main/webapp/WEB-INF/xsd/web-app_2_5.xsd";
//        String xsdFileName = "E://WorkFolder/DevTools/ide/Workplace/keter/project/keter-project//web/src/main/webapp/WEB-INF/xsd/web-app_2_5.xsd";
        String xsdFileName = "E://WorkFolder/DevTools/ide/Workplace/keter/project/keter-project/web/target/test-classes/haha/web/xsd/web-app_2_5.xsd";
        validateXMLByXSD(xmlFileName,xsdFileName);
    }

    /**
     * 通过XSD（XML Schema）校验XML
     */
    public static boolean  validateXMLByXSD(String xmlUri, String xsdUri) {
    	xmlUri = xmlUri.replace("file:/", "");
    	xsdUri = xsdUri.replace("file:/", "");
        try {
            //创建默认的XML错误处理器
            XMLErrorHandler errorHandler = new XMLErrorHandler();
            //获取基于 SAX 的解析器的实例
            SAXParserFactory factory = SAXParserFactory.newInstance();
            //解析器在解析时验证 XML 内容。
            factory.setValidating(true);
            //指定由此代码生成的解析器将提供对 XML 名称空间的支持。
            factory.setNamespaceAware(true);  
            //使用当前配置的工厂参数创建 SAXParser 的一个新实例。
            SAXParser parser = factory.newSAXParser();
            //创建一个读取工具
            SAXReader xmlReader = new SAXReader();
            //获取要校验xml文档实例
            Document xmlDocument = (Document) xmlReader.read(new File(xmlUri));
            //设置 XMLReader 的基础实现中的特定属性。核心功能和属性列表可以在 [url]http://sax.sourceforge.net/?selected=get-set[/url] 中找到。
            parser.setProperty(
                    "http://java.sun.com/xml/jaxp/properties/schemaLanguage",
                    "http://www.w3.org/2001/XMLSchema");
            parser.setProperty(
                    "http://java.sun.com/xml/jaxp/properties/schemaSource",
                    "file:" + xsdUri);
            //创建一个SAXValidator校验工具，并设置校验工具的属性
            SAXValidator validator = new SAXValidator(parser.getXMLReader());
            //设置校验工具的错误处理器，当发生错误时，可以从处理器对象中得到错误信息。
            validator.setErrorHandler(errorHandler);
            long t1 = new DateTime().getMillis();
            //校验
            validator.validate(xmlDocument);
            long t2 = new DateTime().getMillis();
            System.out.println(t2-t1);
            
            XMLWriter writer = new XMLWriter(OutputFormat.createPrettyPrint());
            //如果错误信息不为空，说明校验失败，打印错误信息
            if (errorHandler.getErrors().hasContent()) {
                System.out.println("XML文件通过XSD文件校验失败！");
                writer.write(errorHandler.getErrors());
            } else {
                System.out.println("Good! XML文件通过XSD文件校验成功！");
            }
        } catch (Exception ex) {
            System.out.println("XML文件: " + xmlUri + " 通过XSD文件:" + xsdUri + "检验失败。/n原因： " + ex.getMessage());
            ex.printStackTrace();
        }
        return true;
    }
}
